#!/usr/bin/env python3
"""
测试个人助理的记忆功能

运行此脚本可以验证个人助理的基本功能是否正常工作。
"""

import os
import sys
from pathlib import Path

# 添加当前目录到 Python 路径
current_dir = Path(__file__).parent
sys.path.append(str(current_dir))

try:
    from agent import root_agent
    print("✓ 成功导入个人助理代理")
    
    # 检查代理配置
    print(f"✓ 代理名称: {root_agent.name}")
    print(f"✓ 模型: {root_agent.model}")
    print(f"✓ 工具数量: {len(root_agent.tools)}")
    
    # 列出所有工具
    print("\n可用工具:")
    for i, tool in enumerate(root_agent.tools, 1):
        tool_name = getattr(tool, '__name__', str(tool))
        print(f"  {i}. {tool_name}")
    
    print("\n✓ 个人助理配置正确！")
    print("\n要运行助理，请：")
    print("1. 创建 .env 文件（参考 README.md）")
    print("2. 在父目录运行: adk web")
    print("3. 在浏览器中访问 http://localhost:8000")
    
except ImportError as e:
    print(f"✗ 导入错误: {e}")
    print("请确保已安装 google-adk：pip install google-adk")
    sys.exit(1)
    
except Exception as e:
    print(f"✗ 配置错误: {e}")
    sys.exit(1) 